package com.developmentReformProject.utils;

import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;

public class FileUtil {
    /**
     * 文件上传
     * @param file 源文件
     * @return 新文件名
     * @throws Exception
     */
    public static String upload(MultipartFile file, HttpServletRequest request) throws Exception{
        /*
         * 用时间戳给文件重命名，
         * 避免文件重名而覆盖 先获取文件后缀名，即文件类型
         * 新文件名=时间戳+后缀名
         */
        //获取文件名
        String sourceFileName = file.getOriginalFilename();
        //获取文件类型，即后缀名
        String suffix = sourceFileName.substring(sourceFileName.lastIndexOf("."));
        //新文件名=时间戳+后缀名
        String fileName = System.currentTimeMillis() + suffix;
        /*
         * 将文件保存到tomcat/upload目录下 如果不存在该文件夹，创建文件夹
         */
        //String tomcat = System.getProperty("catalina.home");
        String tomcat = FileUtil.class.getClassLoader().getResource("").getPath();
        String path = tomcat + File.separator + "upload";
        System.out.println("“===============”"+path);
        File folder = new File(path);
        if (!folder.exists()) {
            folder.mkdir();
        }
        //保存文件
        String fileFullName=path+File.separator+fileName;
        file.transferTo(new File(fileFullName));
        return fileName;
    }
    /**
     * 文件下载
     * @param filename 下载的文件名
     * @param resp
     * @throws Exception
     */
    public static void download(String filename, HttpServletResponse resp) throws Exception {
        /*
         * 文件保存在tomcat/upload目录下 从保存位置下载文件
         */
        String tomcat = FileUtil.class.getClassLoader().getResource("").getPath();
        String path = tomcat + File.separator + "upload"+ File.separator+filename;
        /*
         * servlet响应流的方式将文件下载到本地 以附件的方式另存文件
         */
        FileInputStream fis = new FileInputStream(path);
        resp.setCharacterEncoding("utf-8");
        /*
         * 以附件的方式另存文件，每次以1KB单位存储文件
         */
        resp.setHeader("Content-Disposition", "attachment; filename=" + filename);
        ServletOutputStream out = resp.getOutputStream();
        byte[] bt = new byte[1024];
        int length = 0;
        while ((length = fis.read(bt)) != -1) {
            out.write(bt, 0, length);
        }
        fis.close();
        out.close();
    }

}
